import xlrd as xlrd

# https://www.cnblogs.com/tynam/p/11204895.html
from recommend import project

if __name__ == '__main__':
    """
    读取excel表，获得用户特征值，排序
    """
    # 打开文件
    data = xlrd.open_workbook("C:\\Users\\yl2028\\Desktop\\ybrain\\enterprise_20191101-20200505.xlsx")

    # 查看工作表
    data.sheet_names()
    print("sheets：" + str(data.sheet_names()))

    # 通过文件名获得工作表,获取工作表1
    table = data.sheet_by_name('Sheet1')
    header = table.row_values(0)
    size = len(header)
    records=[]
    for row_num in range(1, table.nrows):
        record = {}
        for header_index in range(size):
          record.setdefault(header[header_index], table.row_values(row_num)[header_index])
        records.append(record)
    # for item in records:
    #     print("item:%s" % item)
    print("用户参数: %s" % records[0])
    print("用户总数: %s" % len(records))
    print("用户按地区排行:%s" % project.filter_count(records,"地区"))
    print("用户按地区排行:%s" % project.filter_count(records,"服务归属"))
    print("用户按地区排行:%s" % project.filter_count(records,"管理员名称"))
    print("用户按地区排行:%s" % project.filter_count(records,"会议总时长"))
    # 打印data.sheet_names()可发现，返回的值为一个列表，通过对列表索引操作获得工作表1
    # table = data.sheet_by_index(0)

    # 获取行数和列数
    # 行数：table.nrows
    # 列数：table.ncols
    # print("总行数：" + str(table.nrows))
    # print("总列数：" + str(table.ncols))

    # 获取整行的值 和整列的值，返回的结果为数组
    # 整行值：table.row_values(start,end)
    # 整列值：table.col_values(start,end)
    # 参数 start 为从第几个开始打印，
    # end为打印到那个位置结束，默认为none
    # print("整行值：" + str(table.row_values(0)))
    # print("整列值：" + str(table.col_values(1)))

    # 获取某个单元格的值，例如获取B3单元格值
    # cel_B3 = table.cell(3, 2).value
    # print("第三行第二列的值：" + cel_B3)